# TCP 客户端（与机器人聊天）

import socket

def tcp_client():
    # 创建TCP套接字
    tcp_socket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    
    # 连接服务端（本地测试用localhost，端口8000）
    server_addr = ('localhost', 8000)
    tcp_socket.connect(server_addr)
    print("已连接到聊天机器人，开始对话吧（输入bye结束）：")
    
    try:
        while True:
            # 输入要发送的消息
            msg = input("我：")
            # 发送消息（编码为字节）
            tcp_socket.send(msg.encode('utf-8'))
            
            # 若输入bye，退出循环
            if msg.lower() == 'bye':
                break
            
            # 接收机器人回复
            reply = tcp_socket.recv(1024).decode('utf-8')
            print(f"机器人：{reply}")
    finally:
        # 关闭套接字
        tcp_socket.close()
        print("已断开与服务端的连接。")

if __name__ == "__main__":
    tcp_client()